epochs=(40000 50000 60000)
for epoch in ${epochs[@]}; do
  python ../train_and_impute.py --iteration=${epoch}
done
